Hacker rank python solution list comprehension

visit shbcf.ru